###############################################################################
## AUTHOR: ALAN YAN
## DATE LAST UPDATED: 07/26/2020
## PURPOSE: APPENDIX: ANALYZE YOUGOV CONJOINT DATA - EMPLOYED ONLY  
###############################################################################
rm(list = ls())

#### PACKAGES ####
library(pacman)
p_load(tidyverse,
       estimatr,
       broom,
       haven,
       hrbrthemes,
       cjoint,
       formula.tools, 
       DeclareDesign,
       emmeans,
       lmtest,
       cregg,
       gridExtra,
       egg,
       car,
       stargazer)

#### LOAD DATA ####
df <- read_rds("01-yougov-conjoint/01-data/cleaned/yougov-conjoint-cleaned.rds")

#subset to those who are employed or looking for work
df %>%
  filter(r_employment %in% c("Part-time",
                             "Full-time",
                             "Temporarily laid off")) -> df

#### ANALYZE DATA (NO CONTROLS) ####
#Do people prefer firms with these features?
work <- lm(work_binary ~ 
             as.factor(cje_corporate_gov) +
             as.factor(cje_corporate_resp) +
             as.factor(cje_firm_size) +
             as.factor(cje_political_donations) + 
             as.factor(cje_gender_ownership) + 
             as.factor(cje_healthcare) + 
             as.factor(cje_hours) + 
             as.factor(cje_training) + 
             as.factor(cje_location) + 
             as.factor(cje_sick_leave) + 
             as.factor(cje_parental_leave) + 
             as.factor(cje_race_ownership) + 
             as.factor(cje_retirement) + 
             as.factor(cje_income) +
             as.factor(cje_type_of_work) +
             as.factor(cje_union) +
             as.factor(cje_work_from_home) +
             as.factor(cje_team_work) +
             as.factor(cje_work_culture),
           data = df
) 

#Do people believe some firms are better at dealing with complaints?
complaints <- lm(complaints_binary ~ 
                   as.factor(cje_corporate_gov) +
                   as.factor(cje_corporate_resp) +
                   as.factor(cje_firm_size) +
                   as.factor(cje_political_donations) + 
                   as.factor(cje_gender_ownership) + 
                   as.factor(cje_healthcare) + 
                   as.factor(cje_hours) + 
                   as.factor(cje_training) + 
                   as.factor(cje_location) + 
                   as.factor(cje_sick_leave) + 
                   as.factor(cje_parental_leave) + 
                   as.factor(cje_race_ownership) + 
                   as.factor(cje_retirement) + 
                   as.factor(cje_income) +
                   as.factor(cje_type_of_work) +
                   as.factor(cje_union) +
                   as.factor(cje_work_from_home) +
                   as.factor(cje_team_work) +
                   as.factor(cje_work_culture),
                 data = df
) 

#Do people believe they'll have more responsibilties at some firms?
responsibility <- lm(responsibility_binary ~ 
                       as.factor(cje_corporate_gov) +
                       as.factor(cje_corporate_resp) +
                       as.factor(cje_firm_size) +
                       as.factor(cje_political_donations) + 
                       as.factor(cje_gender_ownership) + 
                       as.factor(cje_healthcare) + 
                       as.factor(cje_hours) + 
                       as.factor(cje_training) + 
                       as.factor(cje_location) + 
                       as.factor(cje_sick_leave) + 
                       as.factor(cje_parental_leave) + 
                       as.factor(cje_race_ownership) + 
                       as.factor(cje_retirement) + 
                       as.factor(cje_income) +
                       as.factor(cje_type_of_work) +
                       as.factor(cje_union) +
                       as.factor(cje_work_from_home) +
                       as.factor(cje_team_work) +
                       as.factor(cje_work_culture),
                     data = df
) 

#Do people believe they'll have more power at some firms?
power <- lm(power_binary ~ 
              as.factor(cje_corporate_gov) +
              as.factor(cje_corporate_resp) +
              as.factor(cje_firm_size) +
              as.factor(cje_political_donations) + 
              as.factor(cje_gender_ownership) + 
              as.factor(cje_healthcare) + 
              as.factor(cje_hours) + 
              as.factor(cje_training) + 
              as.factor(cje_location) + 
              as.factor(cje_sick_leave) + 
              as.factor(cje_parental_leave) + 
              as.factor(cje_race_ownership) + 
              as.factor(cje_retirement) + 
              as.factor(cje_income) +
              as.factor(cje_type_of_work) +
              as.factor(cje_union) +
              as.factor(cje_work_from_home) +
              as.factor(cje_team_work) +
              as.factor(cje_work_culture),
            data = df
) 

#### tables S134 to S145 ####
# define dependent variables
dep_var_labels <- c("Prefer to work",
                    "More power",
                    "More responsibilities",
                    "Better handle complaints")
covariate_labels <- stringr::str_remove_all(names(work$coefficients),
                                            "as.factor\\(cje_corporate_gov\\)|as.factor\\(cje_corporate_resp\\)|as.factor\\(cje_firm_size\\)|as.factor\\(cje_political_donations\\)|as.factor\\(cje_gender_ownership\\)|as.factor\\(cje_healthcare\\)|as.factor\\(cje_hours\\)|as.factor\\(cje_training\\)|as.factor\\(cje_location\\)|as.factor\\(cje_sick_leave\\)|as.factor\\(cje_parental_leave\\)|as.factor\\(cje_race_ownership\\)|as.factor\\(cje_retirement\\)|as.factor\\(cje_income\\)|as.factor\\(cje_type_of_work\\)|as.factor\\(cje_union\\)|as.factor\\(cje_work_from_home\\)|as.factor\\(cje_team_work\\)|as.factor\\(cje_work_culture\\)")[-1]

stargazer(
  work,
  power,
  dep.var.labels = dep_var_labels[1:2],
  covariate.labels = covariate_labels,
  se = list(
    starprep(work, stat = "std.error", clusters = df$rid, se_type = "stata")[[1]],
    starprep(power, stat = "std.error", clusters = df$rid, se_type = "stata")[[1]]
  ),
  p = list(
    starprep(work, stat = "p.value", clusters = df$rid, se_type = "stata")[[1]],
    starprep(power, stat = "p.value", clusters = df$rid, se_type = "stata")[[1]]
  ),
  title = "Replicating Figure A14 work and power results",
  label = "tab:main-effects-all-employed-only-work-power"
)

stargazer(
  responsibility,
  complaints,
  dep.var.labels = dep_var_labels[3:4],
  covariate.labels = covariate_labels,
  se = list(
    starprep(responsibility, stat = "std.error", clusters = df$rid, se_type = "stata")[[1]],
    starprep(complaints, stat = "std.error", clusters = df$rid, se_type = "stata")[[1]]
  ),
  p = list(
    starprep(responsibility, stat = "p.value", clusters = df$rid, se_type = "stata")[[1]],
    starprep(complaints, stat = "p.value", clusters = df$rid, se_type = "stata")[[1]]
  ),
  title = "Replicating Figure A14 responsibility and complaint results",
  label = "tab:main-effects-all-employed-only-responsibility-complaint"
)
